make image in python